package com.djhu.es.core;
/**
 * Created by zw on 2017/12/28.
 */

import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import org.elasticsearch.search.SearchHit;

import java.lang.reflect.Type;
import java.util.Map;

/**
 * code is far away from bug with the animal protecting
 * ┏┓　　　┏┓
 * ┏┛┻━━━┛┻┓
 * ┃　　　　　　　┃
 * ┃　　　━　　　┃
 * ┃　┳┛　┗┳　┃
 * ┃　　　　　　　┃
 * ┃　　　┻　　　┃
 * ┃　　　　　　　┃
 * ┗━┓　　　┏━┛
 * 　　┃　　　┃神兽保佑
 * 　　┃　　　┃代码无BUG！
 * 　　┃　　　┗━━━┓
 * 　　┃　　　　　　　┣┓
 * 　　┃　　　　　　　┏┛
 * 　　┗┓┓┏━┳┓┏┛
 * 　　　┃┫┫　┃┫┫
 * 　　　┗┻┛　┗┻┛
 *
 * <p>
 * ---------------------------------
 */
public class MapRowMapper implements RowMapper<Map<String,Object>>{
    private  Gson gson = new Gson();

    @Override
    public Map<String, Object> mapRow(Object object, int rowSet) throws Exception {
        if (object instanceof Map){
            Map<String, Object> map = (Map<String, Object>)object;
            return map;
        }else if(object instanceof SearchHit){
            SearchHit searchHit = (SearchHit)object;
            Map<String, Object> map = searchHit.getSourceAsMap();
            return map;
        }
        else {
            String value =gson.toJson(object);
            Type type = new TypeToken<Map<String, Object>>() {}.getType();
            Map<String, Object> map = gson.fromJson(value,type);
            return map;
        }
    }
}